package com.mokairui.example.demo;


import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

/**
 * @author Mokairui
 * @Description NIO非阻塞通信下的入门案例：服务端开发
 * @Since 2021/6/29
 */
public class Server {
    public static void main(String[] args) throws IOException {
        // 1.获取通道
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        // 2.切换为非阻塞
        serverSocketChannel.configureBlocking(false);
        // 3.绑定连接的端口
        serverSocketChannel.bind(new InetSocketAddress(9999));
        // 4.获取选择器selector
        Selector selector = Selector.open();
        // 5.将通道注册到选择器上,并且开始指定监听事件
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        // 6.使用Selector选择器轮询已经准备好的事件
        while (selector.select() > 0) {
            // 7.获取选择器中的所有注册的通道中已经就绪好的事件
            Iterator<SelectionKey> it = selector.selectedKeys().iterator();
            // 8.开始遍历这些准备好的事件
            while (it.hasNext()) {
                // 提取这个事件
                SelectionKey next = it.next();
                // 9.判断这个事件具体是什么
                if (next.isAcceptable()) {
                    // 10.直接获取当前接入的客户端通道
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    // 11.切换为非阻塞模式
                    socketChannel.configureBlocking(false);
                    // 12.将本客户端通道注册到选择器
                    socketChannel.register(selector, SelectionKey.OP_READ);
                } else {
                    // 13.获取当前选择器上的读就绪事件
                    SocketChannel channel = (SocketChannel) next.channel();
                    // 14.读取数据
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    int len = 0;
                    while ((len = channel.read(buffer)) > 0) {
                        buffer.flip();
                        System.out.println(new String(buffer.array(), 0 , len));
                        buffer.clear();
                    }
                }
                it.remove(); // 处理完毕之后需要移除当前事件
            }
        }
    }
}
